package com.mm.ai.dto.coze;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;

/**
 * url: https://www.coze.cn/docs/developer_guides/create_knowledge_files
 *
 * @Author：tangnaichao
 * @Date：2024/10/18 19:33
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class CreateknowledgeReqDTO {
    private Long dataset_id;
    private List<DocumentBase> document_bases;
    private ChunkStrategy chunk_strategy;

    @Data
    public static class DocumentBase {
        private String name;
        private SourceInfo source_info;
    }

    @Data
    public static class SourceInfo {
        private String file_base64;
        private String file_type;
    }

    @Data
    @NoArgsConstructor
    @AllArgsConstructor
    public static class ChunkStrategy {
        private String separator;
        private Integer max_tokens;
        private boolean remove_extra_spaces;
        private boolean remove_urls_emails;
        private Integer chunk_type;
    }
}





